tensor([[[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.7900, 0.4300],
           [0.4300, 0.7900]],

          [[0.2800, 0.1800],
           [0.1800, 0.2800]]],


         [[[0.2800, 0.1800],
           [0.1800, 0.2800]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.2100, 0.3700],
           [0.3700, 0.2100]],

          [[0.2700, 0.2800],
           [0.2800, 0.2700]]],


         [[[0.2700, 0.2800],
           [0.2800, 0.2700]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.2000],
           [0.2000, 0.0000]],

          [[0.4600, 0.5400],
           [0.5400, 0.4600]]],


         [[[0.4600, 0.5400],
           [0.5400, 0.4600]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)